shared source/wpf/src/host/shimimpl/persistfile.cxx

   1 //+-----------------------------------------------------------------------
   2 //
   3 //  Copyright (c) Microsoft Corporation.  All rights reserved.
   4 //
   5 //  Description:
   6 //     Defines the PersistFile class of PresentationHost
   7 //
   8 //  History:
   9 //     2002/06/12-murrayw
  10 //          Created
  11 //     2007/09/20-[....]
  12 //          Ported Windows->DevDiv. See SourcesHistory.txt.
  13 //
  14 //------------------------------------------------------------------------
  15
  16 #include "Precompiled.hxx"
  17 #include "PersistFile.hxx"
  18 #include "HostShim.hxx"
  19 #include "ShimUtilities.hxx"
  20
  21 //******************************************************************************
  22 //
  23 // CPersistFile::CPersistFile()
  24 //
  25 //******************************************************************************
  26
  27 CPersistFile::CPersistFile(__in CHostShim* pHostShim)
  28 {
  29     m_pHostShim  = pHostShim;
  30 }
  31
  32 //******************************************************************************
  33 //
  34 // CPersistFile::~CPersistFile()
  35 //
  36 //******************************************************************************
  37
  38 CPersistFile::~CPersistFile()
  39 {
  40 }
  41
  42 //******************************************************************************
  43 //
  44 // CPersistFile::QueryInterface()
  45 //
  46 //******************************************************************************
  47
  48 STDMETHODIMP CPersistFile::QueryInterface(REFIID riid, __out LPVOID *ppReturn)
  49 {
  50     return m_pHostShim->QueryInterface(riid, ppReturn);
  51 }
  52
  53 //******************************************************************************
  54 //
  55 // CPersistFile::AddRef()
  56 //
  57 //******************************************************************************
  58
  59 STDMETHODIMP_(ULONG) CPersistFile::AddRef()
  60 {
  61     return m_pHostShim->AddRef();
  62 }
  63
  64 //******************************************************************************
  65 //
  66 // CPersistFile::Release()
  67 //
  68 //******************************************************************************
  69
  70 STDMETHODIMP_(ULONG) CPersistFile::Release()
  71 {
  72     return m_pHostShim->Release();
  73 }
  74
  75 //******************************************************************************
  76 //
  77 // CPersistFile::GetClassID()
  78 //
  79 //******************************************************************************
  80
  81 STDMETHODIMP CPersistFile::GetClassID(__out LPCLSID pClassID)
  82 {
  83     return m_pHostShim->GetClassID(pClassID);
  84 }
  85
  86 //******************************************************************************
  87 //
  88 // CPersistFile::Save()
  89 //
  90 //******************************************************************************
  91
  92 STDMETHODIMP CPersistFile::Save(__in_ecount(MAX_PATH+1) LPCOLESTR, BOOL)
  93 {
  94     // File->Save is enabled/disabled by IOleCommandTarget->QueryStatus like any other menu item
  95     // However, once enabled it calls IPersistFile->Save rather than IOleCommandtarget->Exec.
  96     // To make it work like the other menu items we simply turn around and call Exec ourselves.
  97     HRESULT hr = S_OK;
  98     IOleCommandTarget* pCmdTarget = NULL;
  99     CKHR(QueryInterface(IID_IOleCommandTarget, (VOID**)&pCmdTarget));
 100     CKHR(pCmdTarget->Exec(NULL, OLECMDID_SAVE, NULL, NULL, NULL));
 101
 102 Cleanup:
 103     ReleaseInterface(pCmdTarget);
 104     return hr;
 105 }
 106
 107 //******************************************************************************
 108 //
 109 // CPersistFile::SaveCompleted()
 110 //
 111 //******************************************************************************
 112
 113 STDMETHODIMP CPersistFile::SaveCompleted(LPCOLESTR pwszFile)
 114 {
 115     return E_NOTIMPL;
 116 }
 117
 118 //******************************************************************************
 119 //
 120 // CPersistFile::Load()
 121 //
 122 //    This method is called if file is local.
 123 //
 124 //******************************************************************************
 125
 126 STDMETHODIMP CPersistFile::Load(__in_ecount(MAX_PATH + 1) LPCOLESTR pwszFile, DWORD dwMode)
 127 {
 128     HRESULT     hr          = S_OK;
 129     CString strMimeType;
 130     MimeType    mimeType    = MimeType_Unknown;
 131
 132     EventWriteWpfHostUm_IPersistFileLoad(pwszFile);
 133
 134     m_pHostShim->SetActivationType(FileActivation);
 135
 136     if (m_pHostShim->GetAllowPersistFileLoad() == FALSE && m_pHostShim->IsAllowPersistFileLoadSet())
 137     {
 138         return E_FAIL;
 139     }
 140
 141     CK_PARG(pwszFile);
 142     if (!*pwszFile)
 143         return E_INVALIDARG;
 144
 145     m_pHostShim->SetStartupUri(pwszFile);
 146     m_pHostShim->SetLocalDeploymentManifestPath(pwszFile);
 147
 148     const DWORD SZMIMESIZE_MAX = 128;
 149     WCHAR wzBuffer[SZMIMESIZE_MAX + 1];
 150     DWORD dwCharCount = SZMIMESIZE_MAX;
 151
 152     CKHR(AssocQueryString(
 153         0, // ASSOCF flags,
 154         ASSOCSTR_CONTENTTYPE, // ASSOCSTR str,
 155         PathFindExtension(pwszFile), // LPCTSTR pszAssoc,
 156         0, // LPCTSTR pszExtra,
 157         wzBuffer, // LPTSTR pszOut,
 158         &dwCharCount // DWORD *pcchOut
 159     ));
 160
 161     mimeType = GetMimeTypeFromString(wzBuffer);
 162
 163     m_pHostShim->SetMimeType(mimeType);
 164
 165     // error out if we don't have an application or container
 166     if (!IsValidMimeType(mimeType))
 167     {
 168         CKHR(E_FAIL);
 169     }
 170
 171     // Note: Execute() triggers Watson on any failure.
 172     m_pHostShim->Execute();
 173     hr = S_OK;
 174
 175 Cleanup:
 176     return hr;
 177 }
 178
 179 //******************************************************************************
 180 //
 181 // CPersistFile::IsDirty()
 182 //
 183 //******************************************************************************
 184
 185 STDMETHODIMP CPersistFile::IsDirty()
 186 {
 187     return S_FALSE;
 188 }
 189
 190 //******************************************************************************
 191 //
 192 // CPersistFile::GetCurFile()
 193 //
 194 //******************************************************************************
 195
 196 STDMETHODIMP CPersistFile::GetCurFile(__out LPOLESTR *ppwszOut)
 197 {
 198     return E_NOTIMPL;
 199 }